package com.adong.juc;

import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.ForkJoinTask;
import java.util.concurrent.RecursiveTask;

/**
 * 分支合并框架
 */
public class ForkJoinDemo {
    public static void main(String[] args) throws ExecutionException, InterruptedException {
        // 生成一个计算任务，负责计算0 - 100
        MyTask myTask = new MyTask(0, 100);
        // 创建一个分支合并池
        ForkJoinPool forkJoinPool = new ForkJoinPool();
        // 执行一个任务
        ForkJoinTask forkJoinTask = forkJoinPool.submit(myTask);
        System.out.println(forkJoinTask.get());
    }
}

/**
 * 分支合并 计算 0 + .... + 100
 */
class MyTask extends RecursiveTask<Integer> {
    private static final int ADJAST_VALUE = 10; // 10 标记
    private int begin; // 开始值
    private int end; // 结束值
    private int result; // 返回结果

    public MyTask(int begin, int end) {
        this.begin = begin;
        this.end = end;
    }

    /**
     * ForkJoinTask与一般任务的主要区别在于它需要实现compute方法，
     * 在这个方法里，首先需要判断任务是否足够小，如果足够小就直接执行任务。
     * 如果不足够小，就必须分割成两个子任务，每个子任务在调用fork方法时，又会进入compute方法，看看当前子任务是否需要继续分割成子任务，
     * 如果不需要继续分割，则执行当前子任务并返回结果。使用join方法会等待子任务执行完并得到其结果。
     *
     * @return
     */
    @Override
    protected Integer compute() {
        // 如果结束-开始<= 10 就不拆分, 直接计算
        if ((end - begin) <= ADJAST_VALUE) {
            for (int i = begin; i <= end; i++) {
                result = result + i;
            }
        } else {
            // 就分裂成两个子任务计算
            int mid = (begin + end) / 2;
            MyTask myTask1 = new MyTask(begin, mid);
            MyTask myTask2 = new MyTask(mid + 1, end);
            // 执行子任务
            myTask1.fork();
            myTask2.fork();

            // 等待子任务执行完，并得到其结果组合
            result = myTask1.join() + myTask2.join();
        }
        return result;
    }
}